package com.sfang.common.util.verifyCode;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.InputStream;

public class CaptchaImageUtil {

    /**
     * 对图片进行旋转
     *
     * @param src   被旋转图片
     * @param angel 旋转角度
     * @return 旋转后的图片
     */
    public static BufferedImage Rotate(Image src, int angel) {
        int src_width = src.getWidth(null);
        int src_height = src.getHeight(null);
        // 计算旋转后图片的尺寸
        Rectangle rect_des = CalcRotatedSize(new Rectangle(new Dimension(
                src_width, src_height)), angel);
        BufferedImage res = null;
        res = new BufferedImage(rect_des.width, rect_des.height,
                BufferedImage.TYPE_INT_RGB);
        Graphics2D g2 = res.createGraphics();
        // 进行转换
        g2.translate((rect_des.width - src_width) / 2,
                (rect_des.height - src_height) / 2);
        g2.rotate(Math.toRadians(angel), src_width / 2, src_height / 2);

        g2.drawImage(src, null, null);
        return res;
    }

    /**
     * 计算旋转后的图片
     *
     * @param src   被旋转的图片
     * @param angel 旋转角度
     * @return 旋转后的图片
     */
    public static Rectangle CalcRotatedSize(Rectangle src, int angel) {
        // 如果旋转的角度大于90度做相应的转换
        if (angel >= 90) {
            if (angel / 90 % 2 == 1) {
                int temp = src.height;
                src.height = src.width;
                src.width = temp;
            }
            angel = angel % 90;
        }

        double r = Math.sqrt(src.height * src.height + src.width * src.width) / 2;
        double len = 2 * Math.sin(Math.toRadians(angel) / 2) * r;
        double angel_alpha = (Math.PI - Math.toRadians(angel)) / 2;
        double angel_dalta_width = Math.atan((double) src.height / src.width);
        double angel_dalta_height = Math.atan((double) src.width / src.height);

        int len_dalta_width = (int) (len * Math.cos(Math.PI - angel_alpha
                - angel_dalta_width));
        int len_dalta_height = (int) (len * Math.cos(Math.PI - angel_alpha
                - angel_dalta_height));
        int des_width = src.width + len_dalta_width * 2;
        int des_height = src.height + len_dalta_height * 2;
        return new Rectangle(new Dimension(des_width, des_height));
    }


    public static String xPic(String[] imgs){//横向处理图片
        try {
            /* 1 读取第一张图片*/
            byte[] bOne = VerifyCodeUtils.getBytes(imgs[0]);
            InputStream baisOne = new ByteArrayInputStream(bOne);
            BufferedImage imageFirst = ImageIO.read(baisOne);
            int width = imageFirst.getWidth();// 图片宽度
            int height = imageFirst.getHeight();// 图片高度
            int[] imageArrayFirst = new int[width * height];// 从图片中读取RGB
            imageArrayFirst = imageFirst.getRGB(0, 0, width, height, imageArrayFirst, 0, width);

            /* 1 对第二张图片做相同的处理 */
            byte[] bTwo = VerifyCodeUtils.getBytes(imgs[1]);
            InputStream baisTwo = new ByteArrayInputStream(bTwo);
            BufferedImage imageSecond = ImageIO.read(baisTwo);
            int[] imageArraySecond = new int[width * height];
            imageArraySecond = imageSecond.getRGB(0, 0, width, height, imageArraySecond, 0, width);

            /* 1 对第二张图片做相同的处理 */
            byte[] bThree = VerifyCodeUtils.getBytes(imgs[2]);
            InputStream baisThree = new ByteArrayInputStream(bThree);
            BufferedImage imageThree = ImageIO.read(baisThree);
            int[] imageArrayThree = new int[width * height];
            imageArrayThree = imageThree.getRGB(0, 0, width, height, imageArrayThree, 0, width);

            /* 1 对第二张图片做相同的处理 */
            byte[] bFour = VerifyCodeUtils.getBytes(imgs[3]);
            InputStream baisFour = new ByteArrayInputStream(bFour);
            BufferedImage imageFour = ImageIO.read(baisFour);
            int[] imageArrayFour = new int[width * height];
            imageArrayFour = imageFour.getRGB(0, 0, width, height, imageArrayFour, 0, width);

            // 生成新图片
            BufferedImage imageResult = new BufferedImage(width * 4 , height,BufferedImage.TYPE_INT_RGB);
            imageResult.setRGB(0, 0, width, height, imageArrayFirst, 0, width);// 设置左半部分的RGB
            imageResult.setRGB(width, 0, width, height, imageArraySecond, 0, width);// 设置右半部分的RGB
            imageResult.setRGB(width*2, 0, width, height, imageArrayThree, 0, width);// 设置右半部分的RGB
            imageResult.setRGB(width*3, 0, width, height, imageArrayFour, 0, width);// 设置右半部分的RGB
            return VerifyCodeUtils.getImageBinary(imageResult);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return "";
    }


    //测试类:注意图片的长宽高不一致可能会报错
    public static void main(String[] args) throws Exception {
        File file = new File("C:\\Users\\admin\\Pictures\\恶意广告.png");
        BufferedImage src = ImageIO.read(file);
        //顺时针旋转90度
        BufferedImage des1 = Rotate(src, 0);
        ImageIO.write(des1, "jpg", new File("e:/90.jpg"));

        //输入图片地址
        String[] imgs={"E:/test.jpg","E:/test1.jpg"};

    }


}
